<?php
/**
 * @title 
 * @author start2004
 */

// ini_set("memory_limit", "128M");

// class Solution {
//
//     function find() {
//
//     }
// }

class Solution {

    /**
     * @param Integer[] $ages
     * @return Integer
     */
    function numFriendRequests($ages) {
        /**
         * @since 2021-12-28 统计每个年龄的数量
         */
        $cntArray = [];
        foreach($ages as $age){
            if($age >= 15){
                if(isset($cntArray[$age])){
                    $cntArray[$age] += 1;
                } else {
                    $cntArray[$age] = 1;
                }
            } else {}
        }

        /**
         * @since 2021-12-28 提取年龄
         */
        $ages = array_keys($cntArray);

        /**
         * @since 2021-12-28 排序
         */
        sort($ages);
        $len = count($ages);
        // echo "ages: ", implode(",", $ages), PHP_EOL;

        /**
         * @since 2021-12-28 好友数量
         */
        $result = 0;

        /**
         * @since 2021-12-28 累计数量
         */
        $sumCntArray = [0];
        for($i=0; $i<$len; $i++){
            $sumCntArray[$i+1] = $sumCntArray[$i] + $cntArray[$ages[$i]];
        }
        // echo "sum: ", implode(",", $sumCntArray), PHP_EOL;

        /**
         * @since 2021-12-28 二分查找
         * @since 2021-12-28 x*0.5+7 < y <= x, x至少15
         * x*0.5+7 = 14.5
         */
        for($i=0; $i<$len; $i++){
            /**
             * @since 2021-12-28 年龄
             */
            $age = $ages[$i];

            /**
             * @since 2021-12-28 数量
             */
            $cnt = $cntArray[$age];

            /**
             * @since 2021-12-28 相同年龄的好友申请
             */
            $result += $cnt * ($cnt-1);

            /**
             * @since 2021-12-28 x*0.5+7 < y < x
             */
            $y = $age/2 + 7;
            // echo "i={$i}, age={$age}, cnt={$cnt}, min-age={$y}", PHP_EOL;

            /**
             * @since 2021-12-28 全部符合
             */
            if($ages[0] > $y){
                $e = 0;
            } else {
                /**
                 * @since 2021-12-29 二分查找
                 */
                $s = 0;
                $e = $i;
                while(($e-$s)>1){
                    $mid = ($s+$e)>>1;
                    $val = $ages[$mid];

                    if($val > $y){
                        $e = $mid;
                    } else {
                        $s = $mid;
                    }
                }
            }

            /**
             * @since 2022-01-05 数量
             */
            $result += $cnt * ($sumCntArray[$i] - $sumCntArray[$e]);
            // echo "符合条件位置={$e}, 符合条件数量=", ($sumCntArray[$i]-$sumCntArray[$e]), PHP_EOL;
        }

        /**
         * @return
         */
        return $result;
    }
}

/**
 * @url http://127.0.0.1/leetcode/202112/2021.12.27.php
 */
$datas = [
    [[16,16]], // 2
    // [[16,17,18]], // 2
    // [[20,30,100,110,120]], // 3
];

include_once dirname(__DIR__) . DIRECTORY_SEPARATOR ."xhprof.php";
$xhprof = new Xhprof();
foreach ($datas as $data){
    if(count($data) <= 20){
        var_dump($data);
    } else {
        var_dump(count($data));
    }

    $obj = new Solution();
    $result = $obj->numFriendRequests(...$data);
    // $result = $obj->($xhprof->tree($data));
    // $result = $obj->($xhprof->listNode($data));
    // $result = $obj->find(...$data);
    var_dump($result);
    // if(count($result)<=20){
    //     var_dump($result);
    // } else {
    //     var_dump(count($result));
    // }
    echo str_repeat(PHP_EOL, 3);
    echo str_repeat("<br>", 3);
}

// foreach ($datas as $data){
//     $obj = new $data[0][0](...$data[1][0]);
//
//     for ($i=1; $i<count($data[0]); $i++){
//         $func = $data[0][$i];
//         $param = $data[1][$i];
//         $result = $obj->$func(...$param);
//
//         echo $data[0][$i] ."(\"". implode(",", $data[1][$i]) ."\") ";
//             if($result === true){
//                 echo "True";
//             } elseif($result === false) {
//                 echo "False";
//             } elseif($result === null) {
//                 echo "Null";
//             } elseif(is_array($result)) {
//                 var_dump($result);
//             } else {
//                 echo $result;
//             }
//         echo PHP_EOL;
//     }
//
//     echo str_repeat(PHP_EOL, 3);
// }
$xhprof->end();
